package com.example.leetcode.prcatice;

/**
 * 集团里有 n 名员工，他们可以完成各种各样的工作创造利润。
 *
 * <p>第 i 种工作会产生 profit[i] 的利润，它要求 group[i] 名成员共同参与。如果成员参与了其中一项工作，就不能参与另一项工作。
 *
 * <p>工作的任何至少产生 minProfit 利润的子集称为 盈利计划 。并且工作的成员总数最多为 n 。
 *
 * <p>有多少种计划可以选择？因为答案很大，所以 返回结果模 10^9 + 7 的值。
 *
 * <p>
 *
 * <p>示例 1：
 *
 * <p>输入：n = 5, minProfit = 3, group = [2,2], profit = [2,3] 输出：2 解释：至少产生 3 的利润，该集团可以完成工作 0 和工作 1
 * ，或仅完成工作 1 。 总的来说，有两种计划。
 */
public class Test879 {
  // n = 5, minProfit = 3, group = [2,2], profit = [2,3]
  public int profitableSchemes(int n, int minProfit, int[] group, int[] profit) {

    int[][] dp = new int[n + 1][minProfit + 1];
    for (int i = 0; i <= n; i++) dp[i][0] = 1;

    for (int k = 0; k < group.length; k++) {
      for (int i = n; i >= group[k]; i--) { // 至少需要的人数
        for (int j = minProfit; j >= 0; j--) { // 利润
          dp[i][j] = (dp[i][j] + dp[i - group[k]][Math.max(0, j - profit[k])]) % 1000000007;
        }
      }
    }
    return dp[n][minProfit];
  }

  public int profitableSchemes2(int n, int minProfit, int[] group, int[] profit) {
    int len = group.length, MOD = (int) 1e9 + 7;
    int[][][] dp = new int[len + 1][n + 1][minProfit + 1];
    dp[0][0][0] = 1;
    for (int i = 1; i <= len; i++) {
      int members = group[i - 1], earn = profit[i - 1];
      for (int j = 0; j <= n; j++) {
        for (int k = 0; k <= minProfit; k++) {
          if (j < members) {
            dp[i][j][k] = dp[i - 1][j][k];
          } else {
            dp[i][j][k] = (dp[i - 1][j][k] + dp[i - 1][j - members][Math.max(0, k - earn)]) % MOD;
          }
        }
      }
    }
    int sum = 0;
    for (int j = 0; j <= n; j++) {
      sum = (sum + dp[len][j][minProfit]) % MOD;
    }
    return sum;
  }
}
